package org.jsola.hr.helper;

import org.jsola.common.SafeKit;
import org.jsola.hr.dto.SchemeTemplateItemAddDTO;
import org.jsola.hr.dto.SchemeTemplateItemUpdateDTO;
import org.jsola.hr.entity.SchemeTemplateItemDO;
import org.jsola.hr.query.SchemeTemplateItemQuery;
import org.jsola.hr.service.ISchemeTemplateItemService;
import org.jsola.hr.vo.SchemeTemplateItemVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.math.BigDecimal;
import java.util.List;

import static org.jsola.hr.core.AbstractTest.SITE_ID;



/**
 * 参保方案模板子项
 *
 * @author zhr
 */
@Component
public class SchemeTemplateItemTestHelper extends BaseTestHelper {
    
    @Autowired
    private ISchemeTemplateItemService schemeTemplateItemService;

    /**
     * 生成add测试数据
     * @return SchemeTemplateItemAddDTO
     */
    public SchemeTemplateItemAddDTO getSchemeTemplateItemAddDTO(){
        SchemeTemplateItemAddDTO schemeTemplateItemAddDTO = new SchemeTemplateItemAddDTO();
        schemeTemplateItemAddDTO.setTemplateId(getRandomValue("varchar",32));
        schemeTemplateItemAddDTO.setCardinalityMin(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setCardinalityMax(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setPersentCompany(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setCardinalityDefault(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setPersentPersonal(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setAmountCompany(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setAmountPersonal(SafeKit.getBigDecimal(getRandomValue("decimal",11)));
        schemeTemplateItemAddDTO.setSort(SafeKit.getInteger(getRandomValue("int",11)));
        return schemeTemplateItemAddDTO;
    }


    /**
     * 生成update测试数据
     * @return SchemeTemplateItemUpdateDTO
     */
    public SchemeTemplateItemUpdateDTO getSchemeTemplateItemUpdateDTO(SchemeTemplateItemVO schemeTemplateItemVO){
        return schemeTemplateItemService.selectDOById(schemeTemplateItemVO.getId(), SITE_ID).to(SchemeTemplateItemUpdateDTO.class);
    }

    /**
     * 生成查询数据
     * @return SchemeTemplateItemQuery
     */
    public SchemeTemplateItemQuery getSchemeTemplateItemQuery() {
        SchemeTemplateItemQuery schemeTemplateItemQuery = new SchemeTemplateItemQuery();
        List<SchemeTemplateItemDO> schemeTemplateItemDOList = schemeTemplateItemService.list(new SchemeTemplateItemDO(),SITE_ID);
        if (!CollectionUtils.isEmpty(schemeTemplateItemDOList)) {
            SchemeTemplateItemDO schemeTemplateItemDO = schemeTemplateItemDOList.get(0);
            schemeTemplateItemQuery = schemeTemplateItemDO.to(SchemeTemplateItemQuery.class);
            if (schemeTemplateItemDO.getCardinalityDefault() != null) {
                schemeTemplateItemQuery.setCardinalityDefaultStart(schemeTemplateItemDO.getCardinalityDefault().subtract(new BigDecimal(1)));
                schemeTemplateItemQuery.setCardinalityDefaultEnd(schemeTemplateItemDO.getCardinalityDefault().add(new BigDecimal(1)));
            }
        }
        return schemeTemplateItemQuery;
    }

    /**
     * 保存示例
     * @return SchemeTemplateItemVO
     */
    public SchemeTemplateItemVO saveSchemeTemplateItemExample(){
        SchemeTemplateItemAddDTO addDTO = this.getSchemeTemplateItemAddDTO();
        SchemeTemplateItemVO schemeTemplateItemVO = schemeTemplateItemService.save(addDTO,buildTokenUser());
        return schemeTemplateItemVO;
    }


}

